import numpy as np

factor_A = [6, 1, 2, 7]
factor_B = [2, 4, 8, 9]
factor_C = [1, 5, 6, 2]

return_profit = [0.02, 0.004, 0.005, 0.007]

corr = np.corrcoef(factor_A, return_profit)
print('factor_A:', corr[0][1])

corr = np.corrcoef(factor_B, return_profit)
print('factor_B:', corr[0][1])

corr = np.corrcoef(factor_C, return_profit)
print('factor_C:', corr[0][1])
